'''
数据处理层：实现连接数据库，写数据库，读数据库
'''
import pymysql
host = '127.0.0.1'
port = 3306
user = 'root'
password = 'wjc123456'
database = 'test'
charset = 'utf8'


def get_conn():
    db = pymysql.Connect(host=host, port=port, user=user, password=password, database=database, charset=charset)
    # print('数据库已连接')
    return db

def read_data(sql):
    db = get_conn()
    cursor = db.cursor()
    data = None
    try:
        cursor.execute(sql)
        data =cursor.fetchall()
        print('记录已读取')
    except Exception as e:
        print(e)
    cursor.close()
    db.close()
    return data

def write_data(sql):
    db =get_conn()
    cursor=db.cursor()
    rows=0
    try:
        rows=cursor.execute(sql)
        db.commit()
        print('记录已写入')
    except Exception as e:
        db.rollback()
        print(e)

    cursor.close()
    db.close()
    return rows

# if __name__ =='__main__':
#     # sql ="update users set pwd ='1234'"
#     print(write_data(sql))